package com.gy.ai.agent.platform.siliconflow.entity;

import com.gy.ai.agent.core.BaseResponse;
import lombok.Data;
import lombok.experimental.Accessors;

import java.io.Serializable;
import java.util.List;

/**
 * @author LGY
 * Create by 2025/6/17 15:27
 */
@Data
@Accessors(chain = true)
public class SiliconflowEmbeddingResp extends BaseResponse implements Serializable {

    private String object;

    private String model;

    private List<Embedding> data;

    private Usage usage;

    @Data
    @Accessors(chain = true)
    public static class Embedding implements Serializable {
        private String object;
        private float[] embedding;
        private Integer index;
    }

    @Data
    @Accessors(chain = true)
    public static class Usage implements Serializable {
        private Long prompt_tokens;
        private Long completion_tokens;
        private Long total_tokens;
    }
}
